#!/bin/bash
#SBATCH -J img_gen
#SBATCH -o img_gen.out
#SBATCH -p A100_40G
#SBATCH -N 1
#SBATCH --gres=gpu:1
#SBATCH --cpus-per-task=8


python /home/qlc/model/jbhi/img_gen/run.py \
--max_epochs=150 \
--val_every=10 \
--logdir=/home/qlc/train_log/img_gen/t1n \
--modality=t1n \
--gli_encoder=/home/qlc/train_log/pre_gli/t1n/best_gli_encoder.pt \
--text_encoder=/home/qlc/train_log/pre_text/best_text_encoder.pt \
--img_encoder=/home/qlc/train_log/pre_img/t1n/best_mri_encoder.pt 

python /home/qlc/model/jbhi/img_gen/run.py \
--max_epochs=150 \
--val_every=10 \
--logdir=/home/qlc/train_log/img_gen/t2w \
--modality=t2w \
--gli_encoder=/home/qlc/train_log/pre_gli/t2w/best_gli_encoder.pt \
--text_encoder=/home/qlc/train_log/pre_text/best_text_encoder.pt \
--img_encoder=/home/qlc/train_log/pre_img/t2w/best_mri_encoder.pt 
